package com.qianfeng.zookeeper.lock;

//
//                            _ooOoo_  
//                           o8888888o  
//                           88" . "88  
//                           (| -_- |)  
//                            O\ = /O  
//                        ____/`---'\____  
//                      .   ' \\| |// `.  
//                       / \\||| : |||// \  
//                     / _||||| -:- |||||- \  
//                       | | \\\ - /// | |  
//                     | \_| ''\---/'' | |  
//                      \ .-\__ `-` ___/-. /  
//                   ___`. .' /--.--\ `. . __  
//                ."" '< `.___\_<|>_/___.' >'"".  
//               | | : `- \`.;`\ _ /`;.`/ - ` : | |  
//                 \ \ `-. \_ __\ /__ _/ .-` / /  
//         ======`-.____`-.___\_____/___.-`____.-'======  
//                            `=---='  
//  
//         .............................................  
//                  佛祖镇楼                  BUG辟易  
//          佛曰:  
//                  写字楼里写字间，写字间里程序员；  
//                  程序人员写程序，又拿程序换酒钱。  
//                  酒醒只在网上坐，酒醉还来网下眠；  
//                  酒醉酒醒日复日，网上网下年复年。  
//                  但愿老死电脑间，不愿鞠躬老板前；  
//                  奔驰宝马贵者趣，公交自行程序员。  
//                  别人笑我忒疯癫，我笑自己命太贱；  
//  


import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.CuratorFrameworkFactory;
import org.apache.curator.framework.recipes.locks.InterProcessMutex;
import org.apache.curator.retry.ExponentialBackoffRetry;
import org.junit.Test;

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.concurrent.CountDownLatch;

/**
 * Created by Jackiechan on 2022/2/9/15:59
 *
 * @author Jackiechan
 * @version 1.0
 * @since 1.0
 */
public class ZkLock {
    private static final String SERVER_URL = "192.168.3.131:2181";

    static int count = 10;

    public void decr() {
        count--;
        System.err.println(count);
    }

    @Test
    public void testLock() throws Exception {
        CountDownLatch countDownLatch = new CountDownLatch(1);
        for (int i = 0; i < 10; i++) {
            new Thread() {
                @Override
                public void run() {

                    CuratorFramework framework = CuratorFrameworkFactory.builder().connectString(SERVER_URL)//设置服务器地址
                            .connectionTimeoutMs(10000)
                            .sessionTimeoutMs(60000)
                            .retryPolicy(new ExponentialBackoffRetry(1000, 5))//重试策略, 最多重试5次,每次等1秒后再重试
                            .build();
                    framework.start();//要先启动
                    InterProcessMutex lock = new InterProcessMutex(framework, "/2108lock");//创建锁
                    //稍微等一会,等所有的线程都创建出来
                    try {
                        countDownLatch.await();//等,等待后面的放行
                        lock.acquire();//抢锁
                        decr();
                        SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss SSS");
                        System.err.println(simpleDateFormat.format(new Date()));
                    } catch (Exception e) {
                        e.printStackTrace();
                    } finally {
                        try {
                            if (lock != null) {
                                lock.release();//释放锁
                            }

                        } catch (Exception e) {
                            e.printStackTrace();
                        }
                    }
                }
            }.start();
        }
        Thread.sleep(20000);
        countDownLatch.countDown();//放行
        System.in.read();
    }
}
